numpy array of 2 dimensions indexing and sum